package com.atguigu.tingshu.user.service.impl;

import com.alibaba.fastjson.JSON;
import com.atguigu.tingshu.common.constant.KafkaConstant;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.service.KafkaService;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.bson.types.ObjectId;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.Assert;

import java.math.BigDecimal;
import java.util.*;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"unchecked", "rawtypes"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private KafkaService kafkaService;

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    @Override
    public Map<String, Object> getLatelyTrack(Long userId) {
        //  声明一个map 集合
        Map<String, Object> map = new HashMap<>();
        //  声明查询对象
        Query query = Query.query(Criteria.where("userId").is(userId)).with(Sort.by(Sort.Direction.DESC,"updateTime"));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        Assert.notNull(userListenProcess,"对象为空");
        //  给map集合赋值数据
        map.put("trackId",userListenProcess.getTrackId());
        map.put("albumId",userListenProcess.getAlbumId());
        //  返回数据
        return map;
    }

    @Override
    public void updateListenProcess(UserListenProcessVo userListenProcessVo, Long userId) {
        //	userId trackId 找 UserListenProcess
        Query query = Query.query(Criteria.where("userId").is(userId).and("trackId").is(userListenProcessVo.getTrackId()));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        //	判断
        if (null != userListenProcess) {
            //  更新记录
            userListenProcess.setUpdateTime(new Date());
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
            //  保存数据.
            //  mongoTemplate.save(userListenProcess,MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS,userId));
        } else {
            //  第一次保存数据.
            userListenProcess = new UserListenProcess();
            //  赋值
            //  userListenProcess.setId(UUID.randomUUID().toString());
            //  为了跟MongoDB中的 Id 数据类型一致
            userListenProcess.setId(new ObjectId().toString());
            userListenProcess.setUserId(userId);
            userListenProcess.setAlbumId(userListenProcessVo.getAlbumId());
            userListenProcess.setTrackId(userListenProcessVo.getTrackId());
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
            userListenProcess.setCreateTime(new Date());
            userListenProcess.setUpdateTime(new Date());
        }
        //  保存数据
        mongoTemplate.save(userListenProcess,MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS,userId));

        //  更改声音的播放量 与 专辑的播放量
        //  定义一个key
        String key = "user:track:"+userId;
        //  获取以 用户Id 与 声音Id 两个维度组成的数据。
        Boolean result = this.redisTemplate.opsForValue().getBit(key, userListenProcessVo.getTrackId());
        //  result = true;{说明已经播放过}  result = false; {说明没有播放过}  getbit 24 110 0 setbit 24 110 1
        if (!result){
            //  记录当前这个用户第一次播放这个声音.
            this.redisTemplate.opsForValue().setBit(key,userListenProcessVo.getTrackId(),true);
            //  给这个key设置过期时间
            this.redisTemplate.expire(key,24,TimeUnit.HOURS);
            //  封装数据 TrackStatMqVo
            TrackStatMqVo trackStatMqVo = new TrackStatMqVo();
            //  赋值：防止重复消费消息。
            trackStatMqVo.setBusinessNo(UUID.randomUUID().toString().replaceAll("-",""));
            trackStatMqVo.setAlbumId(userListenProcessVo.getAlbumId());
            trackStatMqVo.setTrackId(userListenProcessVo.getTrackId());
            trackStatMqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
            trackStatMqVo.setCount(1);
            //  发送消息.
            kafkaService.sendMsg(KafkaConstant.QUEUE_TRACK_STAT_UPDATE, JSON.toJSONString(trackStatMqVo));
        }
    }

    //	如何存储这个声音！
    @Override
    public BigDecimal getTrackBreakSecond(Long trackId, Long userId) {
        //	根据userId, 声音Id 获取数据 从MongoDB 获取数据
        //	where userId = ? and trackId = ? and collectionName = ?; userListenProcess_24
        Query query = Query.query(Criteria.where("userId").is(userId).and("trackId").is(trackId));
        //	collectionName = userListenProcess_24
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        //	判断
        if (null != userListenProcess) {
            //	返回时间
            return userListenProcess.getBreakSecond();
        }
        //	默认返回
        return new BigDecimal("0");
    }
}
